#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/ide.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/gpio.h>
#include <linux/cdev.h>
#include <linux/device.h>
#include <linux/of_gpio.h>
#include <linux/semaphore.h>
#include <linux/timer.h>
#include <linux/irq.h>
#include <linux/wait.h>
#include <linux/poll.h>
#include <linux/fs.h>
#include <linux/fcntl.h>
#include <linux/platform_device.h>
#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>

#define LEDDEV_CNT 1 /* 设备号长度 */
#define LEDDEV_NAME "dtsplatled" /* 设备名字 */
#define LEDOFF 0
#define LEDON 1

/* leddev设备结构体 */
struct leddev_dev{
    dev_t devid; /* 设备号 */
    struct cdev cdev; /* cdev */
    struct class *class; /* 类 */
    struct device *device; /* 设备 */
    int major; /* 主设备号 */
    struct device_node *node; /* LED设备节点 */
    int led0; /* LED灯GPIO标号 */
};

struct leddev_dev leddev;  /* led设备 */


static int led_open(struct inode *inode, struct file *filp)
{
    filp->private_data = &leddev; /* 设置私有数据 */
    return 0;
}

static ssize_t led_write(struct file *filep, const char __user *buf, size_t cnt, loff_t *offt)
{
    int ret;
    unsigned char databuf[2];
    unsigned char ledstat;
    struct leddev_dev *dev = filep->private_data;
    
    ret = copy_from_user(databuf, buf, cnt);
    if(ret < 0) {
        printk("kernel write failed!\r\n");
        return -EFAULT;
    }

    ledstat = databuf[0];

    if (ledstat == LEDON )
        gpio_set_value(dev->led0, 0);
    else if (ledstat == LEDOFF)
        gpio_set_value(dev->led0, 1);
    
    return 0;
}

static int led_release(struct inode *node, struct file *filep)
{
    struct leddev_dev *dev = filep->private_data;
    dev = NULL;

    return 0;
}

/* 设备操作函数 */
static struct file_operations led_fops = {
    .owner = THIS_MODULE,
    .open = led_open,
    .write = led_write,
    .release = led_release,
};

/**@description : flatform驱动的probe函数，当驱动与
 *                 设备匹配以后此函数就会执行
 * @param - dev : platform设备 118 * @return : 0，成功;其他负值,失败
*/
static int led_probe(struct platform_device *dev)
{
    printk("led driver and device was matched!\r\n");
    leddev.major = 0;

    /* 1、设置设备号 */
    if (leddev.major) {  //指定了设备号
        leddev.devid = MKDEV(leddev.major, 0);
        register_chrdev_region(leddev.devid, LEDDEV_CNT, LEDDEV_NAME);
    }
    else {  //未指定设备号，由系统自动分配
        alloc_chrdev_region(&leddev.devid, 0, LEDDEV_CNT, LEDDEV_NAME);
        leddev.major = MAJOR(leddev.devid);
    }
    printk("led devid major = %d \r\n", leddev.major);

    /* 2、注册设备 */
    cdev_init(&leddev.cdev, &led_fops);
    cdev_add(&leddev.cdev, leddev.devid, LEDDEV_CNT);

    /* 3、创建类  在 /sys/class/ 目录下生成 LEDDEV_NAME 目录 */
    leddev.class = class_create(THIS_MODULE, LEDDEV_NAME);
    if (IS_ERR(leddev.class)) {
        return PTR_ERR(leddev.class);
    }

    /* 4、创建设备 在 /dev/ 目录下生成 LEDDEV_NAME 文件 */
    leddev.device = device_create(leddev.class, NULL, leddev.devid, NULL, LEDDEV_NAME);
    if (IS_ERR(leddev.device)) {
        return PTR_ERR(leddev.device);
    }

    /* 5、初始化IO */
    leddev.node = of_find_node_by_path("/gpioled");  /* 匹配设备节点 "/gpioled" */
    if (leddev.node == NULL){
        printk("gpioled node nost find!\r\n");
        return -EINVAL;
    }
    
    leddev.led0 = of_get_named_gpio(leddev.node, "led-gpios", 0);  /* 获取对应的io "led-gpios" */
    if (leddev.led0 < 0) {
        printk("can't get led-gpio\r\n");
        return -EINVAL;
    }
    
    gpio_request(leddev.led0, "led0");  /* 申请IO，并命名为 led0 */
    gpio_direction_output(leddev.led0, 1); /*设置为输出，默认高电平 */

    return 0;
}

/**
 * @description : remove函数，移除platform驱动的时候此函数会执行
 * @param - dev : platform设备
 * @return : 0，成功;其他负值,失败
*/
static int led_remove(struct platform_device *dev)
{
    gpio_set_value(leddev.led0, 1); /* 卸载驱动的时候关闭LED */
    
    cdev_del(&leddev.cdev); /* 删除cdev */
    unregister_chrdev_region(leddev.devid, LEDDEV_CNT);
    device_destroy(leddev.class, leddev.devid);
    class_destroy(leddev.class);
    return 0;
}

/* 匹配列表 */
static const struct of_device_id led_of_match[] = {
    { .compatible = "gpioled" },
    { /* Sentinel */ }
};

MODULE_DEVICE_TABLE(of, led_of_match);

/* platform驱动结构体 */
static struct platform_driver led_driver = {
    .driver = {
        .name = "imx6ul-led", /* 驱动名字，用于和设备匹配 */
        .of_match_table = led_of_match, /* 设备树匹配表 */
    },
    .probe = led_probe,
    .remove = led_remove,
};


/**
 * @description : 驱动模块加载函数
 * @param : 无
 * @return : 无
*/
static int __init leddriver_init(void)
{
    printk("%s\r\n", __func__);
    return platform_driver_register(&led_driver);
}

/**
 * @description : 驱动模块卸载函数
 * @param : 无
 * @return : 无
*/
static void __exit leddriver_exit(void)
{
    printk("%s\r\n", __func__);
    platform_driver_unregister(&led_driver);
}


module_init(leddriver_init);
module_exit(leddriver_exit);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("george <g_yubo@163.com>");
